Ship battle in python 3