Oscar Piastri takes pole position in Bahrain as Lando Norris flounders