add assert statements