adding training for exercise 4
improving exercise 4 csv output