Refactored
*created a menu class