trying to add class for menu (WIP)