updated rep increase in checkpoint
Updated unit tests for models