Slight refactoring, adding downsampling of negative examples