Added json schema generation for models