Not sure if adding shit loads of time on at the end is the best solution, flash a few more yellows and that will sort out the bulk of it. But I reckon time wasting will be massively diminished after a month or two of this. Just needs the consistency from the refs.
I have always believed that diving should be a straight red card with a five match ban. You'd get rid of it in two weeks if you did that.