Added some very basic tests