Extension UI review/improvments