Changes to menu JS