Adding assert statement