Update battleships.py