Basic plotting works, but adding rectangles needs to be fixed