fixed my shitty code