Add slight documentation. Refactor/Cleanup code slightly.