Added comments + bit of cleanup.

Just comments essentially, going over the code again &
re-famililarising.