fix: make sure every plot can be called without ax paramaters (#365)

fix: make sure every plot can be called without ax paramaters (#365) #427