edit prob3 & 4
prob1 add grid
bad figure!