fixes and corrections to LISTA demo.