added a nice potential and plotting goodies