Add return type
Merge pull request #1 from zhongke/master

Add return type