Added some basic functionality tests