Add return
Merge pull request #24 from alicoding/return

Add return