Add more comments throughout

Closes #45. LAST ISSUE!