Wrote a more complete test