Extend up/down migrations to extend the main migrate task