Add version check