Update tantalus.py