Add config for max memory