Battleship game written in python