Add simple assert function