few little comments added
Fixed minor bug.