How to create a Python program which moves