wrote some tests for the new input function