curious kind of discipline you have in mind.

Time-honored punishment: revoke various privileges for periods of time until they get it.

In this case, seems pretty topical to just take the phone away entirely for a few days.