Data science toolkit