Docs

Mostly cleanup to make code look better