Variants of the Hajnal-Szemerédi Theorem