In Timeline view, allow coloring Milestones

Go to a given project and go to timeline view, then options, then layout options, color tasks by and select a custom label… this only color tasks but not Milestones, can you improve this functionality so it also colors milestones?