The problems that you are experiencing look like a minor bug in your code; the code from Aum8 should work ok for any terrain. Well, I was planning to write an article that deals with terrain deformation in the near future so now I've got another reason to do that.
