don't need flatten anymore