Rename `MDS_WIDTH` to `STATE_SIZE`

[richardpringle]

Nit: a more correct description would be to use STATE_SIZE. C.f. https://eprint.iacr.org/2019/458

Originally posted by @dannywillems in #3386 (comment)