During November 2024, Demaross refactored the RockMuonSelection component in the DUNE/ndlar_flow repository to enhance the accuracy of muon track and segment information. Leveraging Python and scientific computing techniques, Demaross introduced a new Z-axis angle calculation and redefined segment identification by deriving segments directly from principal component analysis results. The work included updates to parameter handling and data structures, ensuring more precise capture of muon properties for downstream data analysis and physics simulation. This focused engineering effort established a more robust foundation for muon catalog quality and improved the reliability of subsequent muon-based selection and analysis workflows.