Modified battleship.py