Add missing method
Merge pull request #2 from PunNeng/master

Add missing method