Add screenshots
Organize imports. Rearrange code