This is the second part of my week four assignment