added some stylig
added comment addition
made comment conditional