Added basic edit dataset functionality