Add missing import statements