finishing test code so far