Added import statements for a couple of missing modules.