Finish the one liner pooling layer (only supports max pooling for now)