added missing import and adjusted path to new directory structure