Added missing returns
Merge pull request #4 from bluemath/master

Added missing returns