Adding test bench for the aes key mem module.