Adding support for the sbox