Berkshire's Jain says Geico has made "rapid strides"