added a cross entropy scatter plot