added more input code