compute_cost() fixed, trying to fix gradient_check()