Added code back in to make this level work correctly