Add version checking