Minor improvements. Fixed a typo in comment.