minor fixes + comments added