Worked on the IAS-Astrophysics/athenak repository to enhance scalability and data management for MPI-based simulations by implementing per-rank output files. This feature enables each MPI rank to write its own output, improving both data organization and the resilience of simulation restarts. The approach involved dynamic construction of output paths, updates to restart file handling, and adjustments to output formatting to support the new file structure. Leveraging C++ and Python, the work focused on code refactoring, parallel computing, and robust file I/O. These changes align with high-performance computing best practices and lay the foundation for larger, multi-rank simulation runs.