added plotting back