fixed code for menu