Added per_milestone method and made some changes to the model.