Information Gain Split Finder - Coding problem | OneNoughtOne