Fixed some bugs in state space models, made them use log for fair comparison