added more comments for the imputing part