Update docs and cleanup styles