So made some tweaks to the algorithm with some laplace smoothing and some additional normalization functions... looks better!