Adds support code to start requiring milestones as well as including the
current milestone when generating a new migration.

Update some failing tests

Slight updates