Added more functionality to the logger