Completed assignment for week 5