add  flatten code