battleship in python!