Got migration generator working