new figures
updated figure for test data