Added method return type

https://github.com/kfgodel/diamond/issues/66