Changing logs and fixing import error