Finished writing UnitTests and added unit removal

Units will be removed if they leave the screen, or in a zombies case if
they die