Trying to implement a basic UI