Add assert statement