corrected and added additional training