Use 'sys.stdin.readline()' instead of 'input()'. Add tests