make the code pip friendly