A simple battleship game in python