Quick battleship game in Python