fixed the code to match the directory structure