Working version with simple menu