Added some basic UI