Copied 3D demo
removed unnecessary code
Edited code to show my function
Added back color to show height
Code cleanup