Cleaned up the comments slightly