Updated docs + comments.