Not handling newline at end of input ?

The interact function passes the entire input from stdin, including the new line character. However, your g function expects a string of only two characters.
There are at least two reasonable ways to fix your program:
1. You can remove the \n character from the input before passing it to g: code.
2. You can modify the g function so that it ignores everything except for the first two characters: code.