Updates. Starting to work now. Also test input.