Minor changes to comments. Also noting a better implementation way.