Few updates to the comments and ouputs.