updated step 6, better handle testing of selected model