Update TF Logger functionality