Add peak memory usage support