[x] Add lag to the grid search together with number of states
[ ] Understand how emissions model is fit?
[ ] How to visualize dynamics/ How to visualize learnt parameters? (dynamics show where the system goes with an arrow, because it's linear - vector fields)
[x] Plot inferred states overlaid with trial events like before
[ ] How is prior set?
[ ] Look at initialization - Kmeans;
[ ] check initialization for Poisson HMM
[ ] How can different lags be fit to different features (number of parameters will increase!)
[ ] formally test how to capture two different wheel movement directions with 3-state HMM - model comparison