Cleaned up the UI adding polish.