Battleships game in python