add assert statement