Code review fixes + added docs + fixed some typo in docs.