add restriction function