added draw dataset functionality