add the code from 2 days of traning