Agent penalties

Not the right import, use import unittest instead