This is the second part of my week 04 assignment