Added test files for stages 1 and 2 of the assignment