add test & change some assertTrue to assertEquals