add logging back in