Restrict code working fine, added a text output