Game now ends when all ships are sunk. Restart game also works.