Update castamp.py