Fixed issues with typing and diagnostic plotting #35

Closed Ashar25 closed 11 months ago