UI Modifications - select training model option removed