Added a menu into main.py.