Switch to separate module for DB migration