Added some unit testing stuff