Adding the unit tests to check, fixed minor bug.