fixes #41. uses np.squeeze to remove singleton axis.