Improve comment mode (better cursor position after operation)