Add flax tool version check