A basic battleship game in Python