Video ref and straight red if the player has dived. Then a 3 game ban.

Same punishment for players who scream and swear at the refs, it'll soon stop.