Added more comments.

Need to improve docs!