WIP on week 4 lecture