Added test condition for input