Added comment and removed unsed imports