Fix to allow running apply_rs.py both on data and MC at the same time.

This merge request implements #10 (closed). See the issue for more info about the changes.