Add imports for signals