minor edits + better comments