Problem with python source block

["Doyley, Marvin M." <m.doyley@rochester.edu>]

Hi there,

When I run the following code I get an error, does anybody know how to fix this

I am using the latest version of org-mode.

Thanks,
M


#+BEGIN_SRC python :results file
import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns
x = np.random.rand(100)
y= np.cos(x)
plt.plot(x,y)
plt.savefig('test.png')
return 'test.png'
#+END_SRC

#+RESULTS:
[[file:  File "Org SRC", line 8
    return 'test.png'
    ^
SyntaxError: 'return' outside function
]]