Add extension to migration