Clean up section 5.4 and make the proof much clearer.