Try adding basic UI. WIP