Add simple unittests