How I do replace s-box for present algorithm and put s-box for AES in place in present algorithm

See more:

Python

s-box for present

def sBoxLayer(state):
"""SBox function for encryption

Input: 64-bit integer
Output: 64-bit integer"""

output = 0

for i in range(16):

output += Sbox[( state >> ( i * 4 ) & 0x0F ) ] << ( i * 4 )

What I have tried:

put s-box for AES place s-box for present lightweight algorithm