Fixed a problem with shooting, forward model and state observation