add battleships.py