Add some heck for input