Changed the flatten comment's to make it more understandable