still trying to fix the input issue