Adding source directory as a parameter
Also adding a test directory