Battleship in Python