added tests for the assignment