Small changes, added comments.