Clean up at_ui module