Python script for wargame