small changes before adding the max pooling