Changed from using assertTrue to assertEquals for better testing