Add some basic testing