Cleaned up the python desktop flow example