Finding diameter works