Added some of the initialization -- it seems to be getting to a latent state trajectory, but it still needs the Smoothness penalty