AES code clean

Signed-off-by: David JY Lu <davidlue7@gmail.com>