Add week 3 modifications