Update Battleship game in python