Seperate `class SSM` and `class MambaBlock`.

Seperate `class SSM` and `class MambaBlock`, so that `def __init__` of both classes are clearer. I'm not sure if `class SSM` is supposed to be called `class S6`.