updated final doc with training as well as diagram