Bump version to 0.0.2 and remove module import in setup