Broke down yState function into pieces
Stored timeStep as a object variable instead of passing it in parameters.