Updated for homework in week 6