Fixed plot_simulated_data.py

Fixed what appeared to be a bug in searchlight (removed two lines
with unknown significance).
Fixed docstrings, code clean up.