Have Manage Age Group Reflect in the Mobile App

Manage Age Group has been a great addition to Milestones.
It would be beneficial if the selected age groups were reflected in the Mobile application when selecting Milestones.
Right now, you have to manually select the category when adding a Milestone to an observation activity within the mobile application.