added code for basic UI