Integrate code with main menu

Integrate code