Fixed dropouts bug and correct predict column