js fixes for menu