Made figure much nicer
Changed scale on figure