Add calculation of punishment from prev period