Adding back in cout statements