Edit UI and add new buttons to main menu