Add work model and migration