Merged 2 models into 1 for shared dataset