updated training project code