Add a custom plotting class for the mw problem