Add extra logging capabilities