Simple Menu System in Python