Better swagger doc gen