Add some performance tracking