changed import
tweaked plotting