update README.md

README.md

@@ -2,14 +2,16 @@
 
 Yuxuan Yang, Johannes A. Stork, and Todor Stoyanov
 
-
+## TODO
+- [ ] combine all training models and generating data python scripts to two files(one for position twist representation, the other for position quanternion representation)
+- [ ] add results analyse and plot scripts
+- [ ] python script for animation based on pyBullet (pyBullet setup)
+- [ ] model predictive control script (ROS and Bullet(C++) setup)
 
 ## Demo
 
 ### Long-term prediction
 
-Rollout from our learned model
-
 
 
 ### Model predictive control results