add initial lstm impl