AI As a Thought Partner to Improve Student Writing