wanting to update my code