Clean up logging for demos