Removed Laplace smoothing and also a bug fix for testing