Update ui to make this look nicer.