Adding regeneration functionality