Python file with modified UI for main window