Update python script to support more json files.