Refactor input handler to be compatible with mobile