Small fixes and added comments.