Add missing return to python function
Merge pull request #1 from kyleam/function-return

Add missing return to python function