Add extra logging features