change the way imports the class in python