@saurabh:added model package with 2 classes