Add test for main and organize imports