“Frame” a graph created with draw(_ rect: CGRect)

Question

I want to "frame" a graph created in draw(_ rect: CGRect) (image 1) with a colored box with rounded corners and shadows (image 2).

This is what I h