Small improvements and the introduction of PunctureResponse message