Add typechecking