Working on Menu class in menu.py