created better mock