fixing import error in main