You are replace element before expression with block, and then add whitespace. Why not replace expression with block directly? Also, this works incorrectly for ``if (true)println()``