This is the second part of my week 4 assignment