UI Update to menu - Bug Fixes in life cycle