You will also need to change the unit models to refelct their new abilities and also assign them proper skeletons.