Added initial version of nonlinear dynamics training for the system