slight_change to my code