Modify migration and model. Add spec