Corrected big mistake in KLD implementation
Made changes for the tests for better conventions