Add comments for convolutional and pooling layer