Polish docs based on feedback