Programming assignment week 2