fix the "return NoneType" bug
Merge pull request #56 from hwchiu/fixNoneBug

fix the "return NoneType" bug