Refactored menu to a class.