Modified tests

You could get rid of the `if` below by iterating from 1 instead of 0.