Add mem_size argument