Polish the code and add description.