Battleship written in python