changed params
fix gradient