modified battleship.py