Cole begins rehabilitation