Add support for setup script imports