Probabilistic Gaussian Classifier - Coding problem | OneNoughtOne