Update repo structure for new feature