Adding in some tests.