Update battleship.py