When I change the gridsize from 10x10 to 9x9, I get the following error: self.R[bad_state, :] = self.r_bad IndexError: index 90 is out of bounds for axis 0 with size 82 How can I fix this?