ADDED - 
1. Removed unnecessary imports
2. Fixed all the warning

CLEAN COMMIT. [NO WARNINGS]