Adding correct plot and correcting margin