Grätz, Lukas, Karolinska Institutet, Sweden