Completed assignment for week 2