Modified py package import