Done week 4 programming assignment