diff --git a/lib/mm2lib.py b/lib/mm2lib.py
index 81adb65..f4e1b6b 100644
--- a/lib/mm2lib.py
+++ b/lib/mm2lib.py
@@ -8,29 +8,35 @@
 
 def payment_script(time_lock, secret_hash, pub_0, pub_1):
     """
-    this function making payment script for mm2 atomic swap 
+    this function making payment script for mm2 atomic swap
     ported from mm2 rust code
     """
     return CScript([OP_IF, struct.pack('<I', time_lock), OP_NOP2 , OP_DROP, pub_0, OP_CHECKSIG,
              OP_ELSE, OP_SIZE,  b'\x20', OP_EQUALVERIFY, OP_HASH160, secret_hash, OP_EQUALVERIFY, pub_1, OP_CHECKSIG, OP_ENDIF])
 
 
+def get_payment_address(time_lock, secret_hash, pub_0, pub_1):
+    pubkey_script = payment_script(time_lock, secret_hash, pub_0, pub_1)
+    print(pubkey_script.hex())
+    pubkey_hash = Hash160(pubkey_script)
+    data = b'\x55' + pubkey_hash
+    checksum = sha256(sha256(data).digest()).digest()[:4]
+    byte_address = data + checksum
+    address = bitcoin.base58.encode(byte_address)
+    return address
+
+
 class AddressConvertTest(unittest.TestCase):
 
     def test(self):
 
-        pubkey_script = payment_script(1588875030,
+        address = get_payment_address(1588875030,
                                        bytes.fromhex("bc88c6534d5b82866807cde2da0ce5735c335a2a"),
                                        bytes.fromhex("03683c77e807a47dcd559fa60a6510087e5c5aa0016c094cf5eb4d7e002db18e9f"),
                                        bytes.fromhex("032eadab416e372d21d8cbf798019325088dc796fd6762b6304c0298f279d58038"))
-        print(pubkey_script.hex())
-        pubkey_hash = Hash160(pubkey_script)
-        data = b'\x55' + pubkey_hash
-        checksum = sha256(sha256(data).digest()).digest()[:4]
-        byte_address = data + checksum
-        address = bitcoin.base58.encode(byte_address)
         self.assertEqual(address, "bKYeacaKzVzGEDmrEX6zB5vr2ZoRNh7A3p")
 
 
 if __name__ == '__main__':
     unittest.main()
+