Make UI slightly nicer (#1)