add enhancement for kMath