color scheme and plotting improved