Adding initial code modifications for the PRF-AES