Remove console.log and add UT