Add assert function