Refactor freq waves and add gamma band.
Clean up plotting.