Fixed comment.
Cleaned up code.