Removed some unnecessary code after updating UI design.