Add tests for input