Started adding some minor docs to code