code organizing

nice plots