Diana still loves punishing her man's balls from time to time