First working version of the UI