Added return $this on some methods for chaining