change code to match new json