Adding pot contribution unit testing