WX-78 Skill Tree Ideas YIPPPIEEEE

After the recent lore with WX-78 and Wagstaff finally seeing each other I am incredibly excited for updates on WX-78 and Skill Tree stuff with them. I wanted to hear what you guys think would be a good idea for skill tree features, maybe stuff about losing charge slower or maybe upgrades to Jimmy? 

I kinda wanna see that giant robot idea get implemented but that's wishful thinking because I know that it may have balancing issues, personally I enjoy the robot design that WX-78 rides rather than the weird armor.. unless the design of the armor is changed a bit I don't see myself ever really rooting for it