Updated methods to allow chaining