Final submission for week 3 assignment