Updated code to use new interface with parsed model.