minor changes + comments added