Some minor experimental changes to play with convergence

Changed the step method and made the likelihood function
scaled by the number of data points.
Added plt.close to all report figs to prevent memory leak