add required test code for assignment 1