make things more "pythonic"! (listcomp, flat)