Updates to selection and plotting scripts