Slight function change

Not much at all