Create battleship.py