Add unittests