Add types to migration