swap function in python