Added the missing part absence of which kind of breaks the whole game.

I apologise for that, don't even know how I managed to miss it.