[UI] Make it slightly nicer