Added the expansion function for 16-byte and 32-byte keys.