clean up ui some