set return value instead of undefined
remove unused comment
Merge pull request #4 from ImBobby/develop

Set return value