After working with InputHandler

Signed-off-by: Beansy <nick361@gmail.com>