trying to do a battleship in python