diff --git a/tests/snapshots/flex/test_get_public_key_m/44'/1'/0'_0_0/00000.png b/tests/snapshots/flex/test_get_public_key_m/44'/1'/0'_0_0/00000.png
new file mode 100644
index 00000000..5a1a7890
Binary files /dev/null and b/tests/snapshots/flex/test_get_public_key_m/44'/1'/0'_0_0/00000.png differ
diff --git a/tests/snapshots/flex/test_get_public_key_m/44'/1'/0'_0_1/00000.png b/tests/snapshots/flex/test_get_public_key_m/44'/1'/0'_0_1/00000.png
new file mode 100644
index 00000000..c0086b25
Binary files /dev/null and b/tests/snapshots/flex/test_get_public_key_m/44'/1'/0'_0_1/00000.png differ
diff --git a/tests/snapshots/flex/test_get_public_key_m/44'/1'/0'_0_1/00001.png b/tests/snapshots/flex/test_get_public_key_m/44'/1'/0'_0_1/00001.png
new file mode 100644
index 00000000..b89d3ab1
Binary files /dev/null and b/tests/snapshots/flex/test_get_public_key_m/44'/1'/0'_0_1/00001.png differ
diff --git a/tests/snapshots/flex/test_get_public_key_m/44'/1'/0'_0_1/00002.png b/tests/snapshots/flex/test_get_public_key_m/44'/1'/0'_0_1/00002.png
new file mode 100644
index 00000000..ebd1846f
Binary files /dev/null and b/tests/snapshots/flex/test_get_public_key_m/44'/1'/0'_0_1/00002.png differ
diff --git a/tests/snapshots/flex/test_get_public_key_m/44'/1'/0'_0_2/00000.png b/tests/snapshots/flex/test_get_public_key_m/44'/1'/0'_0_2/00000.png
new file mode 100644
index 00000000..4321e601
Binary files /dev/null and b/tests/snapshots/flex/test_get_public_key_m/44'/1'/0'_0_2/00000.png differ
diff --git a/tests/snapshots/flex/test_get_public_key_m/44'/1'/10'_0_0/00000.png b/tests/snapshots/flex/test_get_public_key_m/44'/1'/10'_0_0/00000.png
new file mode 100644
index 00000000..5a1a7890
Binary files /dev/null and b/tests/snapshots/flex/test_get_public_key_m/44'/1'/10'_0_0/00000.png differ
diff --git a/tests/snapshots/flex/test_get_public_key_m/44'/1'/10'_0_1/00000.png b/tests/snapshots/flex/test_get_public_key_m/44'/1'/10'_0_1/00000.png
new file mode 100644
index 00000000..c0086b25
Binary files /dev/null and b/tests/snapshots/flex/test_get_public_key_m/44'/1'/10'_0_1/00000.png differ
diff --git a/tests/snapshots/flex/test_get_public_key_m/44'/1'/10'_0_1/00001.png b/tests/snapshots/flex/test_get_public_key_m/44'/1'/10'_0_1/00001.png
new file mode 100644
index 00000000..f90841bd
Binary files /dev/null and b/tests/snapshots/flex/test_get_public_key_m/44'/1'/10'_0_1/00001.png differ
diff --git a/tests/snapshots/flex/test_get_public_key_m/44'/1'/10'_0_1/00002.png b/tests/snapshots/flex/test_get_public_key_m/44'/1'/10'_0_1/00002.png
new file mode 100644
index 00000000..fa62612f
Binary files /dev/null and b/tests/snapshots/flex/test_get_public_key_m/44'/1'/10'_0_1/00002.png differ
diff --git a/tests/snapshots/flex/test_get_public_key_m/44'/1'/10'_0_2/00000.png b/tests/snapshots/flex/test_get_public_key_m/44'/1'/10'_0_2/00000.png
new file mode 100644
index 00000000..4321e601
Binary files /dev/null and b/tests/snapshots/flex/test_get_public_key_m/44'/1'/10'_0_2/00000.png differ
diff --git a/tests/snapshots/flex/test_get_public_key_m/44'/1'/2'/1/42_0_0/00000.png b/tests/snapshots/flex/test_get_public_key_m/44'/1'/2'/1/42_0_0/00000.png
new file mode 100644
index 00000000..5a1a7890
Binary files /dev/null and b/tests/snapshots/flex/test_get_public_key_m/44'/1'/2'/1/42_0_0/00000.png differ
diff --git a/tests/snapshots/flex/test_get_public_key_m/44'/1'/2'/1/42_0_1/00000.png b/tests/snapshots/flex/test_get_public_key_m/44'/1'/2'/1/42_0_1/00000.png
new file mode 100644
index 00000000..c0086b25
Binary files /dev/null and b/tests/snapshots/flex/test_get_public_key_m/44'/1'/2'/1/42_0_1/00000.png differ
diff --git a/tests/snapshots/flex/test_get_public_key_m/44'/1'/2'/1/42_0_1/00001.png b/tests/snapshots/flex/test_get_public_key_m/44'/1'/2'/1/42_0_1/00001.png
new file mode 100644
index 00000000..85e46f7e
Binary files /dev/null and b/tests/snapshots/flex/test_get_public_key_m/44'/1'/2'/1/42_0_1/00001.png differ
diff --git a/tests/snapshots/flex/test_get_public_key_m/44'/1'/2'/1/42_0_1/00002.png b/tests/snapshots/flex/test_get_public_key_m/44'/1'/2'/1/42_0_1/00002.png
new file mode 100644
index 00000000..e15bba2a
Binary files /dev/null and b/tests/snapshots/flex/test_get_public_key_m/44'/1'/2'/1/42_0_1/00002.png differ
diff --git a/tests/snapshots/flex/test_get_public_key_m/44'/1'/2'/1/42_0_2/00000.png b/tests/snapshots/flex/test_get_public_key_m/44'/1'/2'/1/42_0_2/00000.png
new file mode 100644
index 00000000..4321e601
Binary files /dev/null and b/tests/snapshots/flex/test_get_public_key_m/44'/1'/2'/1/42_0_2/00000.png differ
diff --git a/tests/snapshots/flex/test_get_public_key_m/48'/1'/4'/1'/0/7_0_0/00000.png b/tests/snapshots/flex/test_get_public_key_m/48'/1'/4'/1'/0/7_0_0/00000.png
new file mode 100644
index 00000000..5a1a7890
Binary files /dev/null and b/tests/snapshots/flex/test_get_public_key_m/48'/1'/4'/1'/0/7_0_0/00000.png differ
diff --git a/tests/snapshots/flex/test_get_public_key_m/48'/1'/4'/1'/0/7_0_1/00000.png b/tests/snapshots/flex/test_get_public_key_m/48'/1'/4'/1'/0/7_0_1/00000.png
new file mode 100644
index 00000000..c0086b25
Binary files /dev/null and b/tests/snapshots/flex/test_get_public_key_m/48'/1'/4'/1'/0/7_0_1/00000.png differ
diff --git a/tests/snapshots/flex/test_get_public_key_m/48'/1'/4'/1'/0/7_0_1/00001.png b/tests/snapshots/flex/test_get_public_key_m/48'/1'/4'/1'/0/7_0_1/00001.png
new file mode 100644
index 00000000..f32007e2
Binary files /dev/null and b/tests/snapshots/flex/test_get_public_key_m/48'/1'/4'/1'/0/7_0_1/00001.png differ
diff --git a/tests/snapshots/flex/test_get_public_key_m/48'/1'/4'/1'/0/7_0_1/00002.png b/tests/snapshots/flex/test_get_public_key_m/48'/1'/4'/1'/0/7_0_1/00002.png
new file mode 100644
index 00000000..7b0858e8
Binary files /dev/null and b/tests/snapshots/flex/test_get_public_key_m/48'/1'/4'/1'/0/7_0_1/00002.png differ
diff --git a/tests/snapshots/flex/test_get_public_key_m/48'/1'/4'/1'/0/7_0_2/00000.png b/tests/snapshots/flex/test_get_public_key_m/48'/1'/4'/1'/0/7_0_2/00000.png
new file mode 100644
index 00000000..4321e601
Binary files /dev/null and b/tests/snapshots/flex/test_get_public_key_m/48'/1'/4'/1'/0/7_0_2/00000.png differ
diff --git a/tests/snapshots/flex/test_get_public_key_m/49'/1'/1'/1/3_0_0/00000.png b/tests/snapshots/flex/test_get_public_key_m/49'/1'/1'/1/3_0_0/00000.png
new file mode 100644
index 00000000..5a1a7890
Binary files /dev/null and b/tests/snapshots/flex/test_get_public_key_m/49'/1'/1'/1/3_0_0/00000.png differ
diff --git a/tests/snapshots/flex/test_get_public_key_m/49'/1'/1'/1/3_0_1/00000.png b/tests/snapshots/flex/test_get_public_key_m/49'/1'/1'/1/3_0_1/00000.png
new file mode 100644
index 00000000..c0086b25
Binary files /dev/null and b/tests/snapshots/flex/test_get_public_key_m/49'/1'/1'/1/3_0_1/00000.png differ
diff --git a/tests/snapshots/flex/test_get_public_key_m/49'/1'/1'/1/3_0_1/00001.png b/tests/snapshots/flex/test_get_public_key_m/49'/1'/1'/1/3_0_1/00001.png
new file mode 100644
index 00000000..640745a1
Binary files /dev/null and b/tests/snapshots/flex/test_get_public_key_m/49'/1'/1'/1/3_0_1/00001.png differ
diff --git a/tests/snapshots/flex/test_get_public_key_m/49'/1'/1'/1/3_0_1/00002.png b/tests/snapshots/flex/test_get_public_key_m/49'/1'/1'/1/3_0_1/00002.png
new file mode 100644
index 00000000..18c18d59
Binary files /dev/null and b/tests/snapshots/flex/test_get_public_key_m/49'/1'/1'/1/3_0_1/00002.png differ
diff --git a/tests/snapshots/flex/test_get_public_key_m/49'/1'/1'/1/3_0_2/00000.png b/tests/snapshots/flex/test_get_public_key_m/49'/1'/1'/1/3_0_2/00000.png
new file mode 100644
index 00000000..4321e601
Binary files /dev/null and b/tests/snapshots/flex/test_get_public_key_m/49'/1'/1'/1/3_0_2/00000.png differ
diff --git a/tests/snapshots/flex/test_get_public_key_m/84'/1'/2'/0/10_0_0/00000.png b/tests/snapshots/flex/test_get_public_key_m/84'/1'/2'/0/10_0_0/00000.png
new file mode 100644
index 00000000..d5b76776
Binary files /dev/null and b/tests/snapshots/flex/test_get_public_key_m/84'/1'/2'/0/10_0_0/00000.png differ
diff --git a/tests/snapshots/flex/test_get_public_key_m/84'/1'/2'/0/10_0_0/00001.png b/tests/snapshots/flex/test_get_public_key_m/84'/1'/2'/0/10_0_0/00001.png
new file mode 100644
index 00000000..abbcf4a7
Binary files /dev/null and b/tests/snapshots/flex/test_get_public_key_m/84'/1'/2'/0/10_0_0/00001.png differ
diff --git a/tests/snapshots/flex/test_get_public_key_m/84'/1'/2'/0/10_0_1/00000.png b/tests/snapshots/flex/test_get_public_key_m/84'/1'/2'/0/10_0_1/00000.png
new file mode 100644
index 00000000..4321e601
Binary files /dev/null and b/tests/snapshots/flex/test_get_public_key_m/84'/1'/2'/0/10_0_1/00000.png differ
diff --git a/tests/snapshots/flex/test_get_public_key_m/86'/1'/4'/1/12_0_0/00000.png b/tests/snapshots/flex/test_get_public_key_m/86'/1'/4'/1/12_0_0/00000.png
new file mode 100644
index 00000000..5a1a7890
Binary files /dev/null and b/tests/snapshots/flex/test_get_public_key_m/86'/1'/4'/1/12_0_0/00000.png differ
diff --git a/tests/snapshots/flex/test_get_public_key_m/86'/1'/4'/1/12_0_1/00000.png b/tests/snapshots/flex/test_get_public_key_m/86'/1'/4'/1/12_0_1/00000.png
new file mode 100644
index 00000000..c0086b25
Binary files /dev/null and b/tests/snapshots/flex/test_get_public_key_m/86'/1'/4'/1/12_0_1/00000.png differ
diff --git a/tests/snapshots/flex/test_get_public_key_m/86'/1'/4'/1/12_0_1/00001.png b/tests/snapshots/flex/test_get_public_key_m/86'/1'/4'/1/12_0_1/00001.png
new file mode 100644
index 00000000..aff137d3
Binary files /dev/null and b/tests/snapshots/flex/test_get_public_key_m/86'/1'/4'/1/12_0_1/00001.png differ
diff --git a/tests/snapshots/flex/test_get_public_key_m/86'/1'/4'/1/12_0_1/00002.png b/tests/snapshots/flex/test_get_public_key_m/86'/1'/4'/1/12_0_1/00002.png
new file mode 100644
index 00000000..53a0208f
Binary files /dev/null and b/tests/snapshots/flex/test_get_public_key_m/86'/1'/4'/1/12_0_1/00002.png differ
diff --git a/tests/snapshots/flex/test_get_public_key_m/86'/1'/4'/1/12_0_2/00000.png b/tests/snapshots/flex/test_get_public_key_m/86'/1'/4'/1/12_0_2/00000.png
new file mode 100644
index 00000000..4321e601
Binary files /dev/null and b/tests/snapshots/flex/test_get_public_key_m/86'/1'/4'/1/12_0_2/00000.png differ
diff --git a/tests/snapshots/flex/test_sign_message_0_0/00000.png b/tests/snapshots/flex/test_sign_message_0_0/00000.png
new file mode 100644
index 00000000..d5b76776
Binary files /dev/null and b/tests/snapshots/flex/test_sign_message_0_0/00000.png differ
diff --git a/tests/snapshots/flex/test_sign_message_0_0/00001.png b/tests/snapshots/flex/test_sign_message_0_0/00001.png
new file mode 100644
index 00000000..4da45ae3
Binary files /dev/null and b/tests/snapshots/flex/test_sign_message_0_0/00001.png differ
diff --git a/tests/snapshots/flex/test_sign_message_0_1/00000.png b/tests/snapshots/flex/test_sign_message_0_1/00000.png
new file mode 100644
index 00000000..4321e601
Binary files /dev/null and b/tests/snapshots/flex/test_sign_message_0_1/00000.png differ
diff --git a/tests/snapshots/flex/test_sign_message_1_0/00000.png b/tests/snapshots/flex/test_sign_message_1_0/00000.png
new file mode 100644
index 00000000..26f75545
Binary files /dev/null and b/tests/snapshots/flex/test_sign_message_1_0/00000.png differ
diff --git a/tests/snapshots/flex/test_sign_message_1_0/00001.png b/tests/snapshots/flex/test_sign_message_1_0/00001.png
new file mode 100644
index 00000000..2f2a9c99
Binary files /dev/null and b/tests/snapshots/flex/test_sign_message_1_0/00001.png differ
diff --git a/tests/snapshots/flex/test_sign_message_1_0/00002.png b/tests/snapshots/flex/test_sign_message_1_0/00002.png
new file mode 100644
index 00000000..3fdb4fd2
Binary files /dev/null and b/tests/snapshots/flex/test_sign_message_1_0/00002.png differ
diff --git a/tests/snapshots/flex/test_sign_psbt_singlesig_pkh_1to1_0_0/00000.png b/tests/snapshots/flex/test_sign_psbt_singlesig_pkh_1to1_0_0/00000.png
new file mode 100644
index 00000000..e31f881c
Binary files /dev/null and b/tests/snapshots/flex/test_sign_psbt_singlesig_pkh_1to1_0_0/00000.png differ
diff --git a/tests/snapshots/flex/test_sign_psbt_singlesig_pkh_1to1_0_1/00000.png b/tests/snapshots/flex/test_sign_psbt_singlesig_pkh_1to1_0_1/00000.png
new file mode 100644
index 00000000..1ad645a5
Binary files /dev/null and b/tests/snapshots/flex/test_sign_psbt_singlesig_pkh_1to1_0_1/00000.png differ
diff --git a/tests/snapshots/flex/test_sign_psbt_singlesig_pkh_1to1_0_2/00000.png b/tests/snapshots/flex/test_sign_psbt_singlesig_pkh_1to1_0_2/00000.png
new file mode 100644
index 00000000..684bc401
Binary files /dev/null and b/tests/snapshots/flex/test_sign_psbt_singlesig_pkh_1to1_0_2/00000.png differ
diff --git a/tests/snapshots/flex/test_sign_psbt_singlesig_pkh_1to1_0_3/00000.png b/tests/snapshots/flex/test_sign_psbt_singlesig_pkh_1to1_0_3/00000.png
new file mode 100644
index 00000000..78571a28
Binary files /dev/null and b/tests/snapshots/flex/test_sign_psbt_singlesig_pkh_1to1_0_3/00000.png differ
diff --git a/tests/snapshots/flex/test_sign_psbt_singlesig_sh_wpkh_1to2_0_0/00000.png b/tests/snapshots/flex/test_sign_psbt_singlesig_sh_wpkh_1to2_0_0/00000.png
new file mode 100644
index 00000000..e31f881c
Binary files /dev/null and b/tests/snapshots/flex/test_sign_psbt_singlesig_sh_wpkh_1to2_0_0/00000.png differ
diff --git a/tests/snapshots/flex/test_sign_psbt_singlesig_sh_wpkh_1to2_0_1/00000.png b/tests/snapshots/flex/test_sign_psbt_singlesig_sh_wpkh_1to2_0_1/00000.png
new file mode 100644
index 00000000..eb68ef02
Binary files /dev/null and b/tests/snapshots/flex/test_sign_psbt_singlesig_sh_wpkh_1to2_0_1/00000.png differ
diff --git a/tests/snapshots/flex/test_sign_psbt_singlesig_sh_wpkh_1to2_1_0/00000.png b/tests/snapshots/flex/test_sign_psbt_singlesig_sh_wpkh_1to2_1_0/00000.png
new file mode 100644
index 00000000..1487210d
Binary files /dev/null and b/tests/snapshots/flex/test_sign_psbt_singlesig_sh_wpkh_1to2_1_0/00000.png differ
diff --git a/tests/snapshots/flex/test_sign_psbt_singlesig_sh_wpkh_1to2_1_1/00000.png b/tests/snapshots/flex/test_sign_psbt_singlesig_sh_wpkh_1to2_1_1/00000.png
new file mode 100644
index 00000000..78571a28
Binary files /dev/null and b/tests/snapshots/flex/test_sign_psbt_singlesig_sh_wpkh_1to2_1_1/00000.png differ
diff --git a/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_1to2_0_0/00000.png b/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_1to2_0_0/00000.png
new file mode 100644
index 00000000..e31f881c
Binary files /dev/null and b/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_1to2_0_0/00000.png differ
diff --git a/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_1to2_0_1/00000.png b/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_1to2_0_1/00000.png
new file mode 100644
index 00000000..feb9222b
Binary files /dev/null and b/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_1to2_0_1/00000.png differ
diff --git a/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_1to2_1_0/00000.png b/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_1to2_1_0/00000.png
new file mode 100644
index 00000000..645da9d4
Binary files /dev/null and b/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_1to2_1_0/00000.png differ
diff --git a/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_1to2_1_1/00000.png b/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_1to2_1_1/00000.png
new file mode 100644
index 00000000..78571a28
Binary files /dev/null and b/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_1to2_1_1/00000.png differ
diff --git a/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_1to2_2_0_0/00000.png b/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_1to2_2_0_0/00000.png
new file mode 100644
index 00000000..e31f881c
Binary files /dev/null and b/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_1to2_2_0_0/00000.png differ
diff --git a/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_1to2_2_0_1/00000.png b/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_1to2_2_0_1/00000.png
new file mode 100644
index 00000000..feb9222b
Binary files /dev/null and b/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_1to2_2_0_1/00000.png differ
diff --git a/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_1to2_2_1_0/00000.png b/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_1to2_2_1_0/00000.png
new file mode 100644
index 00000000..645da9d4
Binary files /dev/null and b/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_1to2_2_1_0/00000.png differ
diff --git a/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_1to2_2_1_1/00000.png b/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_1to2_2_1_1/00000.png
new file mode 100644
index 00000000..78571a28
Binary files /dev/null and b/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_1to2_2_1_1/00000.png differ
diff --git a/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_2to2_0_0/00000.png b/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_2to2_0_0/00000.png
new file mode 100644
index 00000000..e31f881c
Binary files /dev/null and b/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_2to2_0_0/00000.png differ
diff --git a/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_2to2_0_1/00000.png b/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_2to2_0_1/00000.png
new file mode 100644
index 00000000..7dbc7163
Binary files /dev/null and b/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_2to2_0_1/00000.png differ
diff --git a/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_2to2_0_2/00000.png b/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_2to2_0_2/00000.png
new file mode 100644
index 00000000..2a2902c5
Binary files /dev/null and b/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_2to2_0_2/00000.png differ
diff --git a/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_2to2_0_3/00000.png b/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_2to2_0_3/00000.png
new file mode 100644
index 00000000..78571a28
Binary files /dev/null and b/tests/snapshots/flex/test_sign_psbt_singlesig_wpkh_2to2_0_3/00000.png differ
diff --git a/tests/snapshots/stax/test_get_public_key_m/44'/1'/0'_0_0/00000.png b/tests/snapshots/stax/test_get_public_key_m/44'/1'/0'_0_0/00000.png
index 00c634ff..538f3218 100644
Binary files a/tests/snapshots/stax/test_get_public_key_m/44'/1'/0'_0_0/00000.png and b/tests/snapshots/stax/test_get_public_key_m/44'/1'/0'_0_0/00000.png differ
diff --git a/tests/snapshots/stax/test_get_public_key_m/44'/1'/0'_0_1/00000.png b/tests/snapshots/stax/test_get_public_key_m/44'/1'/0'_0_1/00000.png
index 1b793845..55a58f0d 100644
Binary files a/tests/snapshots/stax/test_get_public_key_m/44'/1'/0'_0_1/00000.png and b/tests/snapshots/stax/test_get_public_key_m/44'/1'/0'_0_1/00000.png differ
diff --git a/tests/snapshots/stax/test_get_public_key_m/44'/1'/0'_0_1/00001.png b/tests/snapshots/stax/test_get_public_key_m/44'/1'/0'_0_1/00001.png
index 746852ba..24a03524 100644
Binary files a/tests/snapshots/stax/test_get_public_key_m/44'/1'/0'_0_1/00001.png and b/tests/snapshots/stax/test_get_public_key_m/44'/1'/0'_0_1/00001.png differ
diff --git a/tests/snapshots/stax/test_get_public_key_m/44'/1'/0'_0_1/00002.png b/tests/snapshots/stax/test_get_public_key_m/44'/1'/0'_0_1/00002.png
new file mode 100644
index 00000000..dac62118
Binary files /dev/null and b/tests/snapshots/stax/test_get_public_key_m/44'/1'/0'_0_1/00002.png differ
diff --git a/tests/snapshots/stax/test_get_public_key_m/44'/1'/0'_0_2/00000.png b/tests/snapshots/stax/test_get_public_key_m/44'/1'/0'_0_2/00000.png
index 13499fcc..3f906b2b 100644
Binary files a/tests/snapshots/stax/test_get_public_key_m/44'/1'/0'_0_2/00000.png and b/tests/snapshots/stax/test_get_public_key_m/44'/1'/0'_0_2/00000.png differ
diff --git a/tests/snapshots/stax/test_get_public_key_m/44'/1'/10'_0_0/00000.png b/tests/snapshots/stax/test_get_public_key_m/44'/1'/10'_0_0/00000.png
index 00c634ff..538f3218 100644
Binary files a/tests/snapshots/stax/test_get_public_key_m/44'/1'/10'_0_0/00000.png and b/tests/snapshots/stax/test_get_public_key_m/44'/1'/10'_0_0/00000.png differ
diff --git a/tests/snapshots/stax/test_get_public_key_m/44'/1'/10'_0_1/00000.png b/tests/snapshots/stax/test_get_public_key_m/44'/1'/10'_0_1/00000.png
index 5f744aa0..55a58f0d 100644
Binary files a/tests/snapshots/stax/test_get_public_key_m/44'/1'/10'_0_1/00000.png and b/tests/snapshots/stax/test_get_public_key_m/44'/1'/10'_0_1/00000.png differ
diff --git a/tests/snapshots/stax/test_get_public_key_m/44'/1'/10'_0_1/00001.png b/tests/snapshots/stax/test_get_public_key_m/44'/1'/10'_0_1/00001.png
index f82843b8..c6220afd 100644
Binary files a/tests/snapshots/stax/test_get_public_key_m/44'/1'/10'_0_1/00001.png and b/tests/snapshots/stax/test_get_public_key_m/44'/1'/10'_0_1/00001.png differ
diff --git a/tests/snapshots/stax/test_get_public_key_m/44'/1'/10'_0_1/00002.png b/tests/snapshots/stax/test_get_public_key_m/44'/1'/10'_0_1/00002.png
new file mode 100644
index 00000000..7c57c16e
Binary files /dev/null and b/tests/snapshots/stax/test_get_public_key_m/44'/1'/10'_0_1/00002.png differ
diff --git a/tests/snapshots/stax/test_get_public_key_m/44'/1'/10'_0_2/00000.png b/tests/snapshots/stax/test_get_public_key_m/44'/1'/10'_0_2/00000.png
index 13499fcc..3f906b2b 100644
Binary files a/tests/snapshots/stax/test_get_public_key_m/44'/1'/10'_0_2/00000.png and b/tests/snapshots/stax/test_get_public_key_m/44'/1'/10'_0_2/00000.png differ
diff --git a/tests/snapshots/stax/test_get_public_key_m/44'/1'/2'/1/42_0_0/00000.png b/tests/snapshots/stax/test_get_public_key_m/44'/1'/2'/1/42_0_0/00000.png
index 00c634ff..538f3218 100644
Binary files a/tests/snapshots/stax/test_get_public_key_m/44'/1'/2'/1/42_0_0/00000.png and b/tests/snapshots/stax/test_get_public_key_m/44'/1'/2'/1/42_0_0/00000.png differ
diff --git a/tests/snapshots/stax/test_get_public_key_m/44'/1'/2'/1/42_0_1/00000.png b/tests/snapshots/stax/test_get_public_key_m/44'/1'/2'/1/42_0_1/00000.png
index 84d9c832..55a58f0d 100644
Binary files a/tests/snapshots/stax/test_get_public_key_m/44'/1'/2'/1/42_0_1/00000.png and b/tests/snapshots/stax/test_get_public_key_m/44'/1'/2'/1/42_0_1/00000.png differ
diff --git a/tests/snapshots/stax/test_get_public_key_m/44'/1'/2'/1/42_0_1/00001.png b/tests/snapshots/stax/test_get_public_key_m/44'/1'/2'/1/42_0_1/00001.png
index 1d4e1de7..e84bf60f 100644
Binary files a/tests/snapshots/stax/test_get_public_key_m/44'/1'/2'/1/42_0_1/00001.png and b/tests/snapshots/stax/test_get_public_key_m/44'/1'/2'/1/42_0_1/00001.png differ
diff --git a/tests/snapshots/stax/test_get_public_key_m/44'/1'/2'/1/42_0_1/00002.png b/tests/snapshots/stax/test_get_public_key_m/44'/1'/2'/1/42_0_1/00002.png
new file mode 100644
index 00000000..878411cf
Binary files /dev/null and b/tests/snapshots/stax/test_get_public_key_m/44'/1'/2'/1/42_0_1/00002.png differ
diff --git a/tests/snapshots/stax/test_get_public_key_m/44'/1'/2'/1/42_0_2/00000.png b/tests/snapshots/stax/test_get_public_key_m/44'/1'/2'/1/42_0_2/00000.png
index 13499fcc..3f906b2b 100644
Binary files a/tests/snapshots/stax/test_get_public_key_m/44'/1'/2'/1/42_0_2/00000.png and b/tests/snapshots/stax/test_get_public_key_m/44'/1'/2'/1/42_0_2/00000.png differ
diff --git a/tests/snapshots/stax/test_get_public_key_m/48'/1'/4'/1'/0/7_0_0/00000.png b/tests/snapshots/stax/test_get_public_key_m/48'/1'/4'/1'/0/7_0_0/00000.png
index 00c634ff..538f3218 100644
Binary files a/tests/snapshots/stax/test_get_public_key_m/48'/1'/4'/1'/0/7_0_0/00000.png and b/tests/snapshots/stax/test_get_public_key_m/48'/1'/4'/1'/0/7_0_0/00000.png differ
diff --git a/tests/snapshots/stax/test_get_public_key_m/48'/1'/4'/1'/0/7_0_1/00000.png b/tests/snapshots/stax/test_get_public_key_m/48'/1'/4'/1'/0/7_0_1/00000.png
index c623aa7a..55a58f0d 100644
Binary files a/tests/snapshots/stax/test_get_public_key_m/48'/1'/4'/1'/0/7_0_1/00000.png and b/tests/snapshots/stax/test_get_public_key_m/48'/1'/4'/1'/0/7_0_1/00000.png differ
diff --git a/tests/snapshots/stax/test_get_public_key_m/48'/1'/4'/1'/0/7_0_1/00001.png b/tests/snapshots/stax/test_get_public_key_m/48'/1'/4'/1'/0/7_0_1/00001.png
index a5d6f394..aa4adfea 100644
Binary files a/tests/snapshots/stax/test_get_public_key_m/48'/1'/4'/1'/0/7_0_1/00001.png and b/tests/snapshots/stax/test_get_public_key_m/48'/1'/4'/1'/0/7_0_1/00001.png differ
diff --git a/tests/snapshots/stax/test_get_public_key_m/48'/1'/4'/1'/0/7_0_1/00002.png b/tests/snapshots/stax/test_get_public_key_m/48'/1'/4'/1'/0/7_0_1/00002.png
new file mode 100644
index 00000000..e06d6335
Binary files /dev/null and b/tests/snapshots/stax/test_get_public_key_m/48'/1'/4'/1'/0/7_0_1/00002.png differ
diff --git a/tests/snapshots/stax/test_get_public_key_m/48'/1'/4'/1'/0/7_0_2/00000.png b/tests/snapshots/stax/test_get_public_key_m/48'/1'/4'/1'/0/7_0_2/00000.png
index 13499fcc..3f906b2b 100644
Binary files a/tests/snapshots/stax/test_get_public_key_m/48'/1'/4'/1'/0/7_0_2/00000.png and b/tests/snapshots/stax/test_get_public_key_m/48'/1'/4'/1'/0/7_0_2/00000.png differ
diff --git a/tests/snapshots/stax/test_get_public_key_m/49'/1'/1'/1/3_0_0/00000.png b/tests/snapshots/stax/test_get_public_key_m/49'/1'/1'/1/3_0_0/00000.png
index 00c634ff..538f3218 100644
Binary files a/tests/snapshots/stax/test_get_public_key_m/49'/1'/1'/1/3_0_0/00000.png and b/tests/snapshots/stax/test_get_public_key_m/49'/1'/1'/1/3_0_0/00000.png differ
diff --git a/tests/snapshots/stax/test_get_public_key_m/49'/1'/1'/1/3_0_1/00000.png b/tests/snapshots/stax/test_get_public_key_m/49'/1'/1'/1/3_0_1/00000.png
index 9039782d..55a58f0d 100644
Binary files a/tests/snapshots/stax/test_get_public_key_m/49'/1'/1'/1/3_0_1/00000.png and b/tests/snapshots/stax/test_get_public_key_m/49'/1'/1'/1/3_0_1/00000.png differ
diff --git a/tests/snapshots/stax/test_get_public_key_m/49'/1'/1'/1/3_0_1/00001.png b/tests/snapshots/stax/test_get_public_key_m/49'/1'/1'/1/3_0_1/00001.png
index 97d95bf2..11533ffc 100644
Binary files a/tests/snapshots/stax/test_get_public_key_m/49'/1'/1'/1/3_0_1/00001.png and b/tests/snapshots/stax/test_get_public_key_m/49'/1'/1'/1/3_0_1/00001.png differ
diff --git a/tests/snapshots/stax/test_get_public_key_m/49'/1'/1'/1/3_0_1/00002.png b/tests/snapshots/stax/test_get_public_key_m/49'/1'/1'/1/3_0_1/00002.png
new file mode 100644
index 00000000..5b557219
Binary files /dev/null and b/tests/snapshots/stax/test_get_public_key_m/49'/1'/1'/1/3_0_1/00002.png differ
diff --git a/tests/snapshots/stax/test_get_public_key_m/49'/1'/1'/1/3_0_2/00000.png b/tests/snapshots/stax/test_get_public_key_m/49'/1'/1'/1/3_0_2/00000.png
index 13499fcc..3f906b2b 100644
Binary files a/tests/snapshots/stax/test_get_public_key_m/49'/1'/1'/1/3_0_2/00000.png and b/tests/snapshots/stax/test_get_public_key_m/49'/1'/1'/1/3_0_2/00000.png differ
diff --git a/tests/snapshots/stax/test_get_public_key_m/84'/1'/2'/0/10_0_0/00000.png b/tests/snapshots/stax/test_get_public_key_m/84'/1'/2'/0/10_0_0/00000.png
index 00c634ff..0d6d65ba 100644
Binary files a/tests/snapshots/stax/test_get_public_key_m/84'/1'/2'/0/10_0_0/00000.png and b/tests/snapshots/stax/test_get_public_key_m/84'/1'/2'/0/10_0_0/00000.png differ
diff --git a/tests/snapshots/stax/test_get_public_key_m/84'/1'/2'/0/10_0_0/00001.png b/tests/snapshots/stax/test_get_public_key_m/84'/1'/2'/0/10_0_0/00001.png
index 0ba005e1..ed29648d 100644
Binary files a/tests/snapshots/stax/test_get_public_key_m/84'/1'/2'/0/10_0_0/00001.png and b/tests/snapshots/stax/test_get_public_key_m/84'/1'/2'/0/10_0_0/00001.png differ
diff --git a/tests/snapshots/stax/test_get_public_key_m/84'/1'/2'/0/10_0_1/00000.png b/tests/snapshots/stax/test_get_public_key_m/84'/1'/2'/0/10_0_1/00000.png
index 13499fcc..3f906b2b 100644
Binary files a/tests/snapshots/stax/test_get_public_key_m/84'/1'/2'/0/10_0_1/00000.png and b/tests/snapshots/stax/test_get_public_key_m/84'/1'/2'/0/10_0_1/00000.png differ
diff --git a/tests/snapshots/stax/test_get_public_key_m/86'/1'/4'/1/12_0_0/00000.png b/tests/snapshots/stax/test_get_public_key_m/86'/1'/4'/1/12_0_0/00000.png
index 00c634ff..538f3218 100644
Binary files a/tests/snapshots/stax/test_get_public_key_m/86'/1'/4'/1/12_0_0/00000.png and b/tests/snapshots/stax/test_get_public_key_m/86'/1'/4'/1/12_0_0/00000.png differ
diff --git a/tests/snapshots/stax/test_get_public_key_m/86'/1'/4'/1/12_0_1/00000.png b/tests/snapshots/stax/test_get_public_key_m/86'/1'/4'/1/12_0_1/00000.png
index 4ff8395e..55a58f0d 100644
Binary files a/tests/snapshots/stax/test_get_public_key_m/86'/1'/4'/1/12_0_1/00000.png and b/tests/snapshots/stax/test_get_public_key_m/86'/1'/4'/1/12_0_1/00000.png differ
diff --git a/tests/snapshots/stax/test_get_public_key_m/86'/1'/4'/1/12_0_1/00001.png b/tests/snapshots/stax/test_get_public_key_m/86'/1'/4'/1/12_0_1/00001.png
index 7df57e9e..4786788f 100644
Binary files a/tests/snapshots/stax/test_get_public_key_m/86'/1'/4'/1/12_0_1/00001.png and b/tests/snapshots/stax/test_get_public_key_m/86'/1'/4'/1/12_0_1/00001.png differ
diff --git a/tests/snapshots/stax/test_get_public_key_m/86'/1'/4'/1/12_0_1/00002.png b/tests/snapshots/stax/test_get_public_key_m/86'/1'/4'/1/12_0_1/00002.png
new file mode 100644
index 00000000..5d9f15f7
Binary files /dev/null and b/tests/snapshots/stax/test_get_public_key_m/86'/1'/4'/1/12_0_1/00002.png differ
diff --git a/tests/snapshots/stax/test_get_public_key_m/86'/1'/4'/1/12_0_2/00000.png b/tests/snapshots/stax/test_get_public_key_m/86'/1'/4'/1/12_0_2/00000.png
index 13499fcc..3f906b2b 100644
Binary files a/tests/snapshots/stax/test_get_public_key_m/86'/1'/4'/1/12_0_2/00000.png and b/tests/snapshots/stax/test_get_public_key_m/86'/1'/4'/1/12_0_2/00000.png differ
diff --git a/tests/snapshots/stax/test_sign_message_0_0/00000.png b/tests/snapshots/stax/test_sign_message_0_0/00000.png
index 00c634ff..0d6d65ba 100644
Binary files a/tests/snapshots/stax/test_sign_message_0_0/00000.png and b/tests/snapshots/stax/test_sign_message_0_0/00000.png differ
diff --git a/tests/snapshots/stax/test_sign_message_0_0/00001.png b/tests/snapshots/stax/test_sign_message_0_0/00001.png
index 6b7c4de4..475185ed 100644
Binary files a/tests/snapshots/stax/test_sign_message_0_0/00001.png and b/tests/snapshots/stax/test_sign_message_0_0/00001.png differ
diff --git a/tests/snapshots/stax/test_sign_message_0_1/00000.png b/tests/snapshots/stax/test_sign_message_0_1/00000.png
index 13499fcc..3f906b2b 100644
Binary files a/tests/snapshots/stax/test_sign_message_0_1/00000.png and b/tests/snapshots/stax/test_sign_message_0_1/00000.png differ
diff --git a/tests/snapshots/stax/test_sign_message_1_0/00000.png b/tests/snapshots/stax/test_sign_message_1_0/00000.png
index 25b48b6b..a78662e3 100644
Binary files a/tests/snapshots/stax/test_sign_message_1_0/00000.png and b/tests/snapshots/stax/test_sign_message_1_0/00000.png differ
diff --git a/tests/snapshots/stax/test_sign_message_1_0/00001.png b/tests/snapshots/stax/test_sign_message_1_0/00001.png
index bbf4f05a..a08f0e8b 100644
Binary files a/tests/snapshots/stax/test_sign_message_1_0/00001.png and b/tests/snapshots/stax/test_sign_message_1_0/00001.png differ
diff --git a/tests/snapshots/stax/test_sign_message_1_0/00002.png b/tests/snapshots/stax/test_sign_message_1_0/00002.png
index 45566d4e..97ecbfea 100644
Binary files a/tests/snapshots/stax/test_sign_message_1_0/00002.png and b/tests/snapshots/stax/test_sign_message_1_0/00002.png differ
diff --git a/tests/snapshots/stax/test_sign_psbt_singlesig_pkh_1to1_0_0/00000.png b/tests/snapshots/stax/test_sign_psbt_singlesig_pkh_1to1_0_0/00000.png
index 46daae63..9c787171 100644
Binary files a/tests/snapshots/stax/test_sign_psbt_singlesig_pkh_1to1_0_0/00000.png and b/tests/snapshots/stax/test_sign_psbt_singlesig_pkh_1to1_0_0/00000.png differ
diff --git a/tests/snapshots/stax/test_sign_psbt_singlesig_pkh_1to1_0_1/00000.png b/tests/snapshots/stax/test_sign_psbt_singlesig_pkh_1to1_0_1/00000.png
new file mode 100644
index 00000000..1bcae3c7
Binary files /dev/null and b/tests/snapshots/stax/test_sign_psbt_singlesig_pkh_1to1_0_1/00000.png differ
diff --git a/tests/snapshots/stax/test_sign_psbt_singlesig_pkh_1to1_0_2/00000.png b/tests/snapshots/stax/test_sign_psbt_singlesig_pkh_1to1_0_2/00000.png
new file mode 100644
index 00000000..3516b6f0
Binary files /dev/null and b/tests/snapshots/stax/test_sign_psbt_singlesig_pkh_1to1_0_2/00000.png differ
diff --git a/tests/snapshots/stax/test_sign_psbt_singlesig_pkh_1to1_0_3/00000.png b/tests/snapshots/stax/test_sign_psbt_singlesig_pkh_1to1_0_3/00000.png
new file mode 100644
index 00000000..066a141c
Binary files /dev/null and b/tests/snapshots/stax/test_sign_psbt_singlesig_pkh_1to1_0_3/00000.png differ
diff --git a/tests/snapshots/stax/test_sign_psbt_singlesig_sh_wpkh_1to2_0_0/00000.png b/tests/snapshots/stax/test_sign_psbt_singlesig_sh_wpkh_1to2_0_0/00000.png
index 46daae63..9c787171 100644
Binary files a/tests/snapshots/stax/test_sign_psbt_singlesig_sh_wpkh_1to2_0_0/00000.png and b/tests/snapshots/stax/test_sign_psbt_singlesig_sh_wpkh_1to2_0_0/00000.png differ
diff --git a/tests/snapshots/stax/test_sign_psbt_singlesig_sh_wpkh_1to2_0_1/00000.png b/tests/snapshots/stax/test_sign_psbt_singlesig_sh_wpkh_1to2_0_1/00000.png
new file mode 100644
index 00000000..bfdf581d
Binary files /dev/null and b/tests/snapshots/stax/test_sign_psbt_singlesig_sh_wpkh_1to2_0_1/00000.png differ
diff --git a/tests/snapshots/stax/test_sign_psbt_singlesig_sh_wpkh_1to2_1_1/00000.png b/tests/snapshots/stax/test_sign_psbt_singlesig_sh_wpkh_1to2_1_1/00000.png
new file mode 100644
index 00000000..066a141c
Binary files /dev/null and b/tests/snapshots/stax/test_sign_psbt_singlesig_sh_wpkh_1to2_1_1/00000.png differ
diff --git a/tests/snapshots/stax/test_sign_psbt_singlesig_wpkh_1to2_0_0/00000.png b/tests/snapshots/stax/test_sign_psbt_singlesig_wpkh_1to2_0_0/00000.png
index 46daae63..9c787171 100644
Binary files a/tests/snapshots/stax/test_sign_psbt_singlesig_wpkh_1to2_0_0/00000.png and b/tests/snapshots/stax/test_sign_psbt_singlesig_wpkh_1to2_0_0/00000.png differ
diff --git a/tests/snapshots/stax/test_sign_psbt_singlesig_wpkh_1to2_0_1/00000.png b/tests/snapshots/stax/test_sign_psbt_singlesig_wpkh_1to2_0_1/00000.png
new file mode 100644
index 00000000..74918453
Binary files /dev/null and b/tests/snapshots/stax/test_sign_psbt_singlesig_wpkh_1to2_0_1/00000.png differ
diff --git a/tests/snapshots/stax/test_sign_psbt_singlesig_wpkh_1to2_1_1/00000.png b/tests/snapshots/stax/test_sign_psbt_singlesig_wpkh_1to2_1_1/00000.png
new file mode 100644
index 00000000..066a141c
Binary files /dev/null and b/tests/snapshots/stax/test_sign_psbt_singlesig_wpkh_1to2_1_1/00000.png differ
diff --git a/tests/snapshots/stax/test_sign_psbt_singlesig_wpkh_1to2_2_0_0/00000.png b/tests/snapshots/stax/test_sign_psbt_singlesig_wpkh_1to2_2_0_0/00000.png
index 46daae63..9c787171 100644
Binary files a/tests/snapshots/stax/test_sign_psbt_singlesig_wpkh_1to2_2_0_0/00000.png and b/tests/snapshots/stax/test_sign_psbt_singlesig_wpkh_1to2_2_0_0/00000.png differ
diff --git a/tests/snapshots/stax/test_sign_psbt_singlesig_wpkh_1to2_2_0_1/00000.png b/tests/snapshots/stax/test_sign_psbt_singlesig_wpkh_1to2_2_0_1/00000.png
new file mode 100644
index 00000000..74918453
Binary files /dev/null and b/tests/snapshots/stax/test_sign_psbt_singlesig_wpkh_1to2_2_0_1/00000.png differ
diff --git a/tests/snapshots/stax/test_sign_psbt_singlesig_wpkh_1to2_2_1_1/00000.png b/tests/snapshots/stax/test_sign_psbt_singlesig_wpkh_1to2_2_1_1/00000.png
new file mode 100644
index 00000000..066a141c
Binary files /dev/null and b/tests/snapshots/stax/test_sign_psbt_singlesig_wpkh_1to2_2_1_1/00000.png differ
diff --git a/tests/snapshots/stax/test_sign_psbt_singlesig_wpkh_2to2_0_0/00000.png b/tests/snapshots/stax/test_sign_psbt_singlesig_wpkh_2to2_0_0/00000.png
index 46daae63..9c787171 100644
Binary files a/tests/snapshots/stax/test_sign_psbt_singlesig_wpkh_2to2_0_0/00000.png and b/tests/snapshots/stax/test_sign_psbt_singlesig_wpkh_2to2_0_0/00000.png differ
diff --git a/tests/snapshots/stax/test_sign_psbt_singlesig_wpkh_2to2_0_1/00000.png b/tests/snapshots/stax/test_sign_psbt_singlesig_wpkh_2to2_0_1/00000.png
new file mode 100644
index 00000000..1ee94503
Binary files /dev/null and b/tests/snapshots/stax/test_sign_psbt_singlesig_wpkh_2to2_0_1/00000.png differ
diff --git a/tests/snapshots/stax/test_sign_psbt_singlesig_wpkh_2to2_0_2/00000.png b/tests/snapshots/stax/test_sign_psbt_singlesig_wpkh_2to2_0_2/00000.png
new file mode 100644
index 00000000..8aadd0e2
Binary files /dev/null and b/tests/snapshots/stax/test_sign_psbt_singlesig_wpkh_2to2_0_2/00000.png differ
diff --git a/tests/snapshots/stax/test_sign_psbt_singlesig_wpkh_2to2_0_3/00000.png b/tests/snapshots/stax/test_sign_psbt_singlesig_wpkh_2to2_0_3/00000.png
new file mode 100644
index 00000000..066a141c
Binary files /dev/null and b/tests/snapshots/stax/test_sign_psbt_singlesig_wpkh_2to2_0_3/00000.png differ