Adding tests

should this be in the repo?