Using the unittest library for testing