Fixed Train Model
Updated Wrapper w/ new Train Model methods
Fixed Train Model