Make js more general!
Change margin for h2