Small cleanup of battleship.py