fixes to anneal.py