Can now run Q-controller. Added fancy plotting