* Extend ReturnStatement case to better handle comments in argument 🎉. * Update test. * Add missing space between return and left parenthese. * Fix current tests and add new tests. * Do not break one-liner returns. * Revert specific test. * Patch from #683.