DEV: updated script to generate WM atlas