got first two tests to pass