Add required module imports to controllers