changed UI slightly