Prince Yadav also chipped in with two wickets before walking off with a suspected left hamstring strain