Remove useless imports and added some comments