#20 Adding regression tests once the code was not tested