Updated model handling

Signed-off-by: Brandyn A. White <bwhite@dappervision.com>