Python is fun with math