Corrections to model

Signed-off-by: bhill2 <bjsrhill@gmail.com>