tidy up ast2json.py