Attempt at UnitTesting the original source code