Update method slightly