added return
Merge pull request #3 from topik/ret_changes

added return