Check add_round_key and subbytes functions