Traitor Legion Customization

I'm having a blast playing the game, just completed the campaign and now I'm diving into the side-missions and PVP.

I am a big chaos space marines fan (specifically Iron Warriors) and I loved the customization support for traitor legions in the first game. I'm really hoping that the devs significantly expand & improve the customization available for heretic astartes. Currently it's kind of terrible if I'm being honest. Feels weird to get a downgrade compared to a game that's so much older.

What do y'all think about the currently available options for traitor legions? I really don't like being forced into Night Lords assault armor, or Death Guard bulwark.