Remove puts from testing