Added tests
we love tests