Do you mean run the key through pbkdf2 before using it as the key on the AES-256 stage?