Simple text based game in python