use numpy's exp and pi, arrange the subplots better