Added Context classes, specifict to one training problem