Added some basic testing