Update swap.py