add assert code