Only apply formatting changes to the newly inserted block

Fixes #12 

Note that the use of `WithStatements` ensures that the trivia associated with the block is preserved in the transformation.