changed the p_grid and b_grid. Maybe this will make the optimization more stable