Updates to training project