Cloning now copies initial weights