Updated comments and removed extra couts