Updating setDistortion to allow method chaining