Working on simple menu implementation