Updated json schema generator to use new API