updating my code