Added pot contribution removal: implementation and unit testing