Added cleaned up version of getClassWeight