Updated for Lesson 3 prompt. Bugs exist