2D Plot functionality and possible improvements