Week 2 programming assignment part 1