Fixing small mistakes in code / commentary.