Cleaning up migration generator command