Changed fields in DistantSupervision from private to protected and created UnitDistantSupervision which overrides the run method.