Add victory and game over screens