Add missing imports.
Add missing parameter $environment.