Add ability to update shifts