Code for Fizz Buzz written in Python