First sort

There is a more elegant way to swap variables in Python. Find it and change the code.