HW1 commit
Modified test class