Add update_account! method with fixes