Updates to the code.

I'm forgetting something dumb