Updated fragment shader to show height via multi-step gradient