some fun coding with Python