fix a few issues with 3rd lesson