Let's use the unittest library? Also no need for _test in the file name?