Added a basic UI