fixing all imports and test01 setup