Minor addition

Added a check for restricted shell