Add test units
Update package.json