Add more comments and clean up the code a bit.