Wrote some tests