[CLEANUP] Organize imports.
[TASK] Add the Unit tests to the IDEA setup.