Fixed my shitty programming