added confidence score,  fixed importing bug