Add some edge cases

Note the last two new cases are fail now.

There are also some cases which I am not sure which is the "right" solution, pls see the comments.
just ignore it, since this project doesn't aim to help fix styles issues other than semicolons.