Add game over code for lesson 4