Hi,
the TCP handshake is there, but it's not that easy to isolate it and add it to UDP.
Your idea can be implemented, but it's not the easiest thing in the world.
I'd suggest to start with the tutorial and focus on the UDP applications (your idea could be implemented at application layer). You could have some issues in reading / setting the ECN bits (I never used them). If you find such a limitation ask for help, probably we'll need to extend ns-3 and add some functions.
Cheers,
T.