Small improvements - UI css etc.