changed the directory to be correct and added tests (no problems were modified after the deadline, you can check the codes)