Putting it in Python