bug fixes. e2e working.