Update warlock.py