Added necessary packages for plotting and fixed Register Comparison