Update mockingjay.py