Cleaned up menu btn code