changed the shitty code again