UI setup cleanup