Add arguments to the plot function

Accepted arguments are: linear, semilogy, semilogx, lolog