Drawing on top of UIView (Swift)

Question

I have a view, wishing which, a custom view draws a graph by overriding draw(_ rect: CGRect) (image 1).

I want that graph to be "framed" by a color