Removed plot

Removed plot from end of function.