Fix comments.

Minor change. Removed unused import.