Some basic python code that I want to play with