Create battleship.py

Run using python battleship.py A or B