Add better comments

Clean up the comments in the code, since they were out of date.