Make sure semicolon is inserted after `return`

Closes #2
Merge pull request #3 from bfrohs/patch-1

Thanks @bfrohs