Add create_group function