cleaned up the UI