This is first and second part of my week four assignment