Add shell initialization