Remove prints add logging