Update notation to match lecture notes