Added puts to the last couple lines