fix: make sure every plot can be called without ax paramaters

fix: make sure every plot can be called without ax paramaters #426