Swedish gov't to take direct stake in new nuclear reactors