Added contain method and updated test