Some progress, but still issues not displaying final puts if certain conditions are met.