Simple pendulum in Python!!