Fixed my fucked up code