Billy has a tantrum and dumps puzzle pieces on the floor. Billy not only has to pick up the puzzle pieces, but he also has to clean up the entire toy room. Which punishment procedure involves restoring the environment to a condition better than it was prior to the problem behavior