Added some comments. Small changes.