Updated the import statements of a couple files

Made the imports work from the project root instead
of from the cwd.