Create Battleship game in python