Add some edits to the milestone, minor method rename in test.