Update method working now