Add python shell kwarg to locate module