shape is the second arg in your parameter list in the doc string. See my not above for the `__init__` method to solve this.