Adding functions for wp menu