Add nd.squeeze function, with tests