Python battleship program