Removed unused import. Added some comments.