Added ability to specify JSON models