Add some documentation if possible

[yonzarecki]

For future students working on this, we probably need to add more documentation to our code.