Updating styles and docs per suggestions in the PR