Updated UI

Added a little bit more :) Still under construction.