Implemented the basic UI