Basic battleship game in python