Ship battle in python3