Update from print to logger class