Fixes lesson 3