- Clean up Ui class