Writing An Algorithm with Python