The problem is there isn't enough incentive for players to stay on their feet. If they go down theatrically the worst that can happen is a freekick to the opponents in their own box.
I'd like that, if a dive can be proved, the decision is reversed. I.e. a penalty to the other team. Yellow cards for diving should also carry across games, so if you are caught diving twice in a season you automatically get a ban.