Fix imports.
Add setup script.