added more amit edits to paper