Fixes dual plotting issue in hw1-a