Made ui nicer