clean code to plot policies
