added code for menu