Update the POM example's shader code for correct height-finding algorithm