Daniel
You can buy 4mm square key stock...but if ya wanna do it like i have...get a tatty old 6mm allen key and grind of 2 opposite points to the depth of the motor gearbox....instant drive shaft !!!

Cheers Froggy