added remove dataset behaviour