Epic comment! Also improved design in here I think