Simple Battleship game w/ ai in python