Cleaning up ui