Add -i flag for input