Added seperate class for menu