Modified the structure. Added a menu class