Add the unittest module