how to write to a json file python