final code changes for week 2
final commit for week 02