Add command restriction