UI update

Added some basic ui for Jonas to play around with