small tweak and added comments