Add support for method chaining