Add logic for active class in menu