change assert to assert_equal and added solution for 2.5