MEGADETH - "insomnia" Tablature

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|-----------|

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