Add assert statements