Added some more comments.
Small improvements.