Refactor test organization.
Add basic main structure.