made fixes to inf theory agent