add puts
remove if