ship battle in python