commented code a bit more, editted our write up a bit as well