Update init_account method