plot in 3d space with points and customise colour for different points

Question

I wrote the code below for plotting in 3D space. I want colour red for the (x=0,y=0,z=1) and yellow for (x=0,y=0,z=1) point and the rest unchanged. and I want to ad grid lin