Add method chaining support