Make sure semicolon is inserted after `return`

Closes #2