Simple roguelike in Python with Pygame