Adjusted UI slightly