WIP for adding main menu