Exercise: Write some tests II