Add input handling. Still buggy though :(