Added additional comments to improve readability

Meant both for us, and future people working on this project