Bagnaia looking to improve with bike