Added missing import statements