Add assert.contains function