Update new-migration template to support custom Migration type

Related to: #220 (closed)

This MR updates the template used by the new migration script to account for the changes make in this MR: !390 (merged)