Add more logging capabilities