Removed the ID column because I ended up not using it at all
Fixed typo on if statement