adding part 2 of exercise 1 - assignment 1