Added main menu functionality