Wrote a few tests