Fix ui.py indentation