Add and rename plotting functions to utils/plots.py