remove puts in code