Update zombies.cpp

Woring correctly for the first test case