Add more comments and clean up code a bit.