Added missing `returns` method. Close #1