fixing errors from week 2