Need to add assertions here