Comments/small edit