Creating the CNN model