Improve comments

Try to comply with Godoc format