assert statements added