Updating my code