Simple Pokémon game in Python