Refactored Migration.up/1 and down/1