E|----------|
B|----------|
G|---2--2---|
D|--0----0--|
A|----------|
E|----------|

Here's how the riff is played at first:
E|-------------------|
B|-------------------|
G|------------2--2---|
D|------0----0----0--| x2
A|--2--2-------------|
E|-------------------|

And then this is the one played for the rest:
E|---------------------|
B|---------------------|
G|--------------2--2---|
D|------0--0---0----0--| then just play this
A|--2--2----2----------| a bunch.
E|---------------------|

Now, skipping to the end of the song,
literally the very end.
E|-------------------|
B|-------------------|
G|-------0-0--2/4-0--|
D|--0-0--2-2------2--|
A|--2-2--------------|
E|-------------------|
Again, just listen for the timing.