Fixed bug by adding puts