Megadeth - Insomnia Chords & Tabs

This should be right, I got it from a book.
The beginning goes like this:
Riff A:
e|-----------|
b|-----------|
g|-----------|
d|-0-0-0-0-0-|
A|-7-7-8-5-7-|
E|-----------|
[ Tab from: https://www.guitartabs.cc/tabs/m/megadeth/insomnia_tab_ver_2.html ]
Then when the distortion kicks in, play Riff A 3 times then add this
riff:
e|-------|
b|-------|
g|-5-4-1-|
d|-5-4-1-|
A|-3-2-3-|
E|-------|

If this isn't right, let me know at clownfish@iname.com