[WIP] Refactor MSM

`fit_method`?