[WIP] Improved frame for the plot