I want to implement same functionality with cnn model along with the step of creating a model with dataset.