Add more comments and clean up the code.