Add type-checking