Update castle-defense.py