Added legend and changed axis to correct