Added procedure to plot omegas, still not working