add documentation to the model class
updating test_size_sample1