added TODO statements to the test file