Implement input module to intercept key input