minor changes - basic UI