added dataset parsing back in