cleared some import, removed the assertEquals method