Python: Battleship code to create board.