The Function Stage. The final step or the rehab process is to work towards restoring function. In order to effectively restore a patient to pre-injury levels of function, it's important to address higher-level capabilities in order to reduce the risk of re-injury.