setstate now follows constructor convention of using momentum for all params, not just matrices.