Tests

Referring to the grading grading rubric for guidance, add any needed tests for your solution.