Adding assert statements