Enhance UI with pythonSelection