Add migration script for assignment driver

Also made this attribute required, to simplify things.