Add peak memory usage measurement capability