Replace assert_equal calls with assert statements
Remove assert_equal function