MSD plotting enhanced