Added code to the main class  for the menu