Fix the number of plotted samples

1. make the number of plotted samples the same as specified
2. make the plot look nice
3. update the save to workspace function - save I & Q