Removed unused import, tweaked comment