The ending in the book is quite different and very ironic
Just saying...
http://en.wikipedia.org/wiki/A_Shock_to_the_System#Difference_between_book_and_film
Can't decide if it would improve the film or not. It's a very interesting twist in any case.